"""
数据模式模块
包含所有API请求和响应的数据模式定义
"""

from .user import UserCreate, UserUpdate, UserResponse
from .conversation import ConversationCreate, ConversationUpdate, ConversationResponse, MessageCreate, MessageResponse
from .model_config import ModelConfigCreate, ModelConfigUpdate, ModelConfigResponse
from .prompt_template import PromptTemplateCreate, PromptTemplateUpdate, PromptTemplateResponse
from .knowledge_base import KnowledgeBaseCreate, KnowledgeBaseUpdate, KnowledgeBaseResponse, DocumentCreate, DocumentResponse
from .chat import ChatRequest, ChatResponse, ChatStreamResponse
from .mcp import (
    MCPSessionCreate, MCPSessionResponse, MCPSessionUpdate,
    MCPContextCreate, MCPContextResponse, MCPContextUpdate,
    MCPMessageCreate, MCPMessageResponse,
    MCPRequestPayload, MCPResponsePayload,
    MCPSessionStats
)

__all__ = [
    "UserCreate", "UserUpdate", "UserResponse",
    "ConversationCreate", "ConversationUpdate", "ConversationResponse",
    "MessageCreate", "MessageResponse",
    "ModelConfigCreate", "ModelConfigUpdate", "ModelConfigResponse",
    "PromptTemplateCreate", "PromptTemplateUpdate", "PromptTemplateResponse",
    "KnowledgeBaseCreate", "KnowledgeBaseUpdate", "KnowledgeBaseResponse",
    "DocumentCreate", "DocumentResponse",
    "ChatRequest", "ChatResponse", "ChatStreamResponse",
    "MCPSessionCreate", "MCPSessionResponse", "MCPSessionUpdate",
    "MCPContextCreate", "MCPContextResponse", "MCPContextUpdate",
    "MCPMessageCreate", "MCPMessageResponse",
    "MCPRequestPayload", "MCPResponsePayload",
    "MCPSessionStats"
]
